# autor: Fabiano A. Lima
# Esse arquivo contém o código usado para gera ras tabelas e estimar os modelos.


tbl_citacoes <- readRDS("./data/citacoes_lj_sm.rds")
folha_sm_sem_lj <- readRDS("./data/folha_sm_sem_lj.rds")
o_globo_sm_sem_lj <- readRDS("./data/o_globo_sm_sem_lj.rds")

tbl_citacoes$Total <- tbl_citacoes$Folha + tbl_citacoes$`O Globo`

gera_tabela_totais(tbl_citacoes)

df_final <- read_csv("./data/lava jato sergio moro 2014-03-17 a 2018-09-30.csv", col_types = "Ddd", na="<1", skip = 2) %>%
  replace_na(list(`Operação Lava Jato: (Brasil)` = 0.1, `Sérgio Moro: (Brasil)` = 0.1)) %>%
  left_join(folha$gera_df(folha_sm_sem_lj, "folha"), by = "Semana") %>%
  left_join(o_globo$gera_df(o_globo_sm_sem_lj, "globo"), by = "Semana") %>%
  replace_na(list(folha = 0, globo = 0)) %>%
  mutate(
    cit_indep = folha + globo, 
    `Ano.2015` = ifelse(year(Semana) == 2015, 1, 0),
    `Ano.2016` = ifelse(year(Semana) == 2016, 1, 0),
    `Ano.2017` = ifelse(year(Semana) == 2017, 1, 0),
    `Ano.2018` = ifelse(year(Semana) == 2018, 1, 0)
  )


# Modelos aninhados
fit_1 <- lm(`Sérgio Moro: (Brasil)` ~ `Operação Lava Jato: (Brasil)`, data = df_final)
fit_2 <- lm(`Sérgio Moro: (Brasil)` ~ `Operação Lava Jato: (Brasil)` + cit_indep, data = df_final)
fit_3 <- lm(`Sérgio Moro: (Brasil)` ~ `Operação Lava Jato: (Brasil)` + cit_indep + `Ano.2015` + `Ano.2016` + `Ano.2017` + `Ano.2018`, data = df_final)


summary(fit_1)
summary(fit_2)
summary(fit_3)

gera_tabela(fit_1, fit_2, fit_3, "2015":"2018", "2014")